package com.yuan.yblog.common.exception;

import cn.hutool.core.util.StrUtil;
import com.yuan.yblog.common.api.CommonResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

import java.sql.SQLSyntaxErrorException;

/**
 * 全局异常捕捉
 */
@Slf4j
@ControllerAdvice
public class GlobalExceptionHandler {
    /**
     * 统一处理 ApiException异常，这个异常是由自定义断言类Assert抛出的
     * 通过CommonResult把结果返回前端
     * @param e 异常对象
     * @return 封装结果CommonResult
     */
    @ResponseBody
    @ExceptionHandler(value = ApiException.class)
    public CommonResult<String> handle(ApiException e) {
        if (e.getResultCode() != null) {
            return CommonResult.failed(e.getResultCode());
        }
        return CommonResult.failed(e.getMessage());
    }

    /**
     * 统一处理SQLSyntaxErrorException异常，这个异常是sql语法错误抛出
     * 抛出这个异常的原因一般是：没启动后台服务
     * @param e 异常对象
     */
    @ResponseBody
    @ExceptionHandler(value = SQLSyntaxErrorException.class)
    public CommonResult<String> handleSQLSyntaxErrorException(SQLSyntaxErrorException e) {
        String message = e.getMessage();
        if (StrUtil.isNotEmpty(message)) {
            // 打印错误
            log.error(e.getMessage());
            message = "数据库操作错误";
        }
        return CommonResult.failed(message);
    }

}
